package com.aishua.fire.config.jwtsecurity.res;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.SecurityMetadataSource;
import org.springframework.security.access.intercept.AbstractSecurityInterceptor;
import org.springframework.security.access.intercept.InterceptorStatusToken;
import org.springframework.security.web.FilterInvocation;
import org.springframework.security.web.access.intercept.FilterInvocationSecurityMetadataSource;
import org.springframework.stereotype.Service;

import com.aishua.fire.common.req.BodyReaderHttpServletRequestWrapper;

import java.io.IOException;

/**
 * Created by yangyibo on 17/1/19.
 */
@Service
public class MyFilterSecurityInterceptor extends AbstractSecurityInterceptor implements Filter {

	private static final String FILTER_APPLIED = "__spring_security_demoFilter_filterApplied";
	
    @Autowired
    private FilterInvocationSecurityMetadataSource securityMetadataSource;

    @Autowired
    public void setMyAccessDecisionManager(MyAccessDecisionManager myAccessDecisionManager) {
        super.setAccessDecisionManager(myAccessDecisionManager);
    }


    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void doFilter(ServletRequest requestInter, ServletResponse response, FilterChain chain) throws IOException, ServletException {
    	HttpServletRequest request = (HttpServletRequest)requestInter;
    	
    	if (request.getAttribute(FILTER_APPLIED) != null) {
    		chain.doFilter(request, response);
            return ;
        }
        //do something
    	request.setAttribute(FILTER_APPLIED,true);
    	
    	/*String [] allowDomain= {"http://localhost:8000","http://xiaochengxu.leanapp.cn","http://hk.bjaishua.com","http://192.168.0.106:8000","http://localhost:8001"};
        Set<String> allowedOrigins= new HashSet<String>(Arrays.asList(allowDomain)); 
        String originHeader= request.getHeader("Origin");
        
        if (allowedOrigins.contains(originHeader)){
        	HttpServletResponse resp = (HttpServletResponse)response;
	        resp.setHeader("Access-Control-Allow-Origin",originHeader);  
	        resp.setHeader("Access-Control-Allow-Credentials", "true");  
	        resp.setHeader("Access-Control-Allow-Methods", "POST, GET, OPTIONS, DELETE,PUT");  
	        resp.setHeader("Access-Control-Max-Age", "3600");  
	        resp.setHeader("Access-Control-Allow-Headers", HeaderConstants.X_REQUESTED_WITH+","+HeaderConstants.AUTHORIZATION+","+HeaderConstants.CONTENT_TYPE);
        }*/
    	ServletRequest requestWrapper = new BodyReaderHttpServletRequestWrapper(request);
        FilterInvocation fi = new FilterInvocation(requestWrapper, response, chain);
        invoke(fi);
    }


    private void invoke(FilterInvocation fi) throws IOException, ServletException {
		//fi里面有一个被拦截的url
		//里面调用MyInvocationSecurityMetadataSource的getAttributes(Object object)这个方法获取fi对应的所有权限
		//再调用MyAccessDecisionManager的decide方法来校验用户的权限是否足够
    	InterceptorStatusToken token = null;
        try {
        	token = super.beforeInvocation(fi);
        	//执行下一个拦截器
            fi.getChain().doFilter(fi.getRequest(),fi.getResponse());
        } finally {
            super.afterInvocation(token, null);
        }
    }

    @Override
    public void destroy() {

    }

    @Override
    public Class<?> getSecureObjectClass() {
        return FilterInvocation.class;
    }

    @Override
    public SecurityMetadataSource obtainSecurityMetadataSource() {
        return this.securityMetadataSource;
    }
}